package cn.zhyjohn.paysystem.common;

import com.swetake.util.Qrcode;
import org.springframework.stereotype.Component;
import org.springframework.util.Base64Utils;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.File;
import java.io.IOException;
import java.io.OutputStream;

/**
 * 
 * 二维码 <br>
 * 创建时间：2016年12月03日
 * 
 * @author JOHN
 *
 */
public class CreateQRCode {
	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param imgPath 图片路径
	 * @throws IOException 
	 */
	public static void encoderQRCode(String content, String imgPath) throws IOException {
		encoderQRCode(content, imgPath, "png", 2);
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param output  输出流
	 * @throws IOException 
	 */
	public static void encoderQRCode(String content, OutputStream output) throws IOException {
		encoderQRCode(content, output, "png", 2);
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param imgPath 图片路径
	 * @param imgType 图片类型
	 * @throws IOException 
	 */
	public static void encoderQRCode(String content, String imgPath, String imgType) throws IOException {
		encoderQRCode(content, imgPath, imgType, 2);
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param output  输出流
	 * @param imgType 图片类型
	 * @throws IOException 
	 */
	public static void encoderQRCode(String content, OutputStream output, String imgType) throws IOException {
		encoderQRCode(content, output, imgType, 2);
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param imgPath 图片路径
	 * @param imgType 图片类型
	 * @param size    二维码尺寸
	 * @throws IOException
	 */
	public static void encoderQRCode(String content, String imgPath, String imgType, int size) throws IOException {

		BufferedImage bufImg = qRCodeCommon(content, imgType, size);

		File imgFile = new File(imgPath);
		if (!imgFile.exists()) {
			imgFile.mkdirs();
		}
		// 生成二维码QRCode图片
		ImageIO.write(bufImg, imgType, imgFile);
	}

	/**
	 * 生成二维码(QRCode)图片
	 * 
	 * @param content 存储内容
	 * @param output  输出流
	 * @param imgType 图片类型
	 * @param size    二维码尺寸
	 * @throws IOException
	 */
	public static void encoderQRCode(String content, OutputStream output, String imgType, int size) throws IOException {
		BufferedImage bufImg = qRCodeCommon(content, imgType, size);
		// 生成二维码QRCode图片
		ImageIO.write(bufImg, imgType, output);
	}

	/**
	 * 生成二维码(QRCode)图片
	 *
	 * @param content 存储内容
	 * @return Base64
	 * @throws IOException 
	 */
	public static String encoderQRCode(String content) throws IOException {
		BufferedImage bufImg = qRCodeCommon(content, "jpg", 7);
		ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
		// 生成二维码QRCode图片
		ImageIO.write(bufImg, "jpg", outputStream);
		return Base64Utils.encodeToString(outputStream.toByteArray());
	}

	/**
	 * 生成二维码(QRCode)图片的公共方法
	 * 
	 * @param content 存储内容
	 * @param imgType 图片类型
	 * @param size    二维码尺寸
	 * @return
	 */
	private static BufferedImage qRCodeCommon(String content, String imgType, int size) {
		BufferedImage bufImg = null;
		size = 7;
		try {
			Qrcode qrcodeHandler = new Qrcode();
			// 设置二维码排错率，可选L(7%)、M(15%)、Q(25%)、H(30%)，排错率越高可存储的信息越少，但对二维码清晰度的要求越小
			qrcodeHandler.setQrcodeErrorCorrect('Q');
			// N代表数字,A代表字符a-Z,B代表其他字符
			qrcodeHandler.setQrcodeEncodeMode('B');
			// 设置设置二维码尺寸，取值范围1-40，值越大尺寸越大，可存储的信息越大
			qrcodeHandler.setQrcodeVersion(size);
			// 获得内容的字节数组，设置编码格式
			byte[] contentBytes = content.getBytes("utf-8");

			boolean[][] codeOut = qrcodeHandler.calQrcode(contentBytes);
			// 图片尺寸 加10是二维码上下左右白边为5个像素
			int imgSize = codeOut.length * size + 10;
			// int imgSize = 67 + 12 * (size - 1);

			bufImg = new BufferedImage(imgSize, imgSize, BufferedImage.TYPE_INT_RGB);
			Graphics2D gs = bufImg.createGraphics();
			// 设置背景颜色
			gs.setBackground(Color.WHITE);
			gs.clearRect(0, 0, imgSize, imgSize);

			// 设定图像颜色> BLACK
			gs.setColor(Color.BLACK);
			// 设置偏移量，不设置可能导致解析出错
			int pixoff = size;

			// 输出内容> 二维码
			if (contentBytes.length > 0 && contentBytes.length < 800) {
				for (int i = 0; i < codeOut.length; i++) {
					for (int j = 0; j < codeOut.length; j++) {
						if (codeOut[j][i]) {
							gs.fillRect(j * pixoff + 5, i * pixoff + 5, size, size);
						}
					}
				}
			} else {
				throw new Exception("QRCode content bytes length = " + contentBytes.length + " not in [0, 800].");
			}
			gs.dispose();
			bufImg.flush();
		} catch (Exception e) {
			e.printStackTrace();
		}
		return bufImg;
	}
}
